from src.model.shufflenet_v2 import ShuffleNet_V2
from src.utils import Data

# 训练模型
if __name__ == '__main__':

    shuffleNet_V2 = ShuffleNet_V2()
    model = shuffleNet_V2.build(input_shape = (256, 256, 3), classes = 4)